Journals
  Publication Years
  Keywords
Search within results Open Search
Please wait a minute...
For Selected: Toggle Thumbnails
De-raining algorithm based on joint attention mechanism for single image
Chengxia XU, Qing YAN, Teng LI, Kaichao MIAO
Journal of Computer Applications    2022, 42 (8): 2578-2585.   DOI: 10.11772/j.issn.1001-9081.2021061072
Abstract321)   HTML21)    PDF (1959KB)(109)       Save

It is challenging for the existing single image de-raining algorithms to fully explore the interaction of attention mechanisms in different dimensions. Therefore, an algorithm based on joint attention mechanism was proposed to realize single image de-raining. The algorithm contains a channel attention mechanism and a spatial attention mechanism. Specifically, in the channel attention mechanism, the distribution of rain streak features in each channel was detected and the importance of each feature channel was differentiated. In the spatial attention mechanism, aiming at the spatial relationship of rain streak distribution within channels, the context information was accumulated in a local to global manner to realize efficient and accurate de-raining. Additionally, a deep residual shrinkage network with a soft threshold nonlinear transformation sub-network embedded in the residual module was used to zero out redundant information via a soft threshold function, thereby improving the ability of the CNN in retaining image details in noise. Experiments were carried out on open rainfall data sets and self constructed rainfall data sets. Compared with spatial attention, the joint attention rain removal algorithm improved Peak Signal-to-Noise Ratio (PSNR) by 4.5% and the Structural SIMilarity (SSIM) by 0.3%. Experimental results show that the proposed algorithm can effectively perform single image de-raining and image detail preserving. At the same time, this algorithm outperforms the comparison algorithms in terms of visual effect and quantitative metrics.

Table and Figures | Reference | Related Articles | Metrics